// Copyright (c) 2012 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "net/http/http_request_headers.h"

#include <memory>

#include "base/values.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace net {

namespace {

    TEST(HttpRequestHeaders, HasHeader)
    {
        HttpRequestHeaders headers;
        headers.SetHeader("Foo", "bar");
        EXPECT_TRUE(headers.HasHeader("foo"));
        EXPECT_TRUE(headers.HasHeader("Foo"));
        EXPECT_FALSE(headers.HasHeader("Fo"));

        const HttpRequestHeaders& headers_ref = headers;
        EXPECT_TRUE(headers_ref.HasHeader("foo"));
        EXPECT_TRUE(headers_ref.HasHeader("Foo"));
        EXPECT_FALSE(headers_ref.HasHeader("Fo"));
    }

    TEST(HttpRequestHeaders, SetHeader)
    {
        HttpRequestHeaders headers;
        headers.SetHeader("Foo", "bar");
        EXPECT_EQ("Foo: bar\r\n\r\n", headers.ToString());
    }

    TEST(HttpRequestHeaders, SetMultipleHeaders)
    {
        HttpRequestHeaders headers;
        headers.SetHeader("Cookie-Monster", "Nom nom nom");
        headers.SetHeader("Domo-Kun", "Loves Chrome");
        EXPECT_EQ("Cookie-Monster: Nom nom nom\r\nDomo-Kun: Loves Chrome\r\n\r\n",
            headers.ToString());
    }

    TEST(HttpRequestHeaders, SetHeaderTwice)
    {
        HttpRequestHeaders headers;
        headers.SetHeader("Foo", "bar");
        headers.SetHeader("Foo", "bar");
        EXPECT_EQ("Foo: bar\r\n\r\n", headers.ToString());
    }

    TEST(HttpRequestHeaders, SetHeaderTwiceCaseInsensitive)
    {
        HttpRequestHeaders headers;
        headers.SetHeader("Foo", "bar");
        headers.SetHeader("FoO", "Bar");
        EXPECT_EQ("Foo: Bar\r\n\r\n", headers.ToString());
    }

    TEST(HttpRequestHeaders, SetHeaderTwiceSamePrefix)
    {
        HttpRequestHeaders headers;
        headers.SetHeader("FooBar", "smokes");
        headers.SetHeader("Foo", "crack");
        EXPECT_EQ("FooBar: smokes\r\nFoo: crack\r\n\r\n", headers.ToString());
        const HttpRequestHeaders& headers_ref = headers;
        EXPECT_EQ("FooBar: smokes\r\nFoo: crack\r\n\r\n", headers_ref.ToString());
    }

    TEST(HttpRequestHeaders, SetEmptyHeader)
    {
        HttpRequestHeaders headers;
        headers.SetHeader("Foo", "Bar");
        headers.SetHeader("Bar", "");
        EXPECT_EQ("Foo: Bar\r\nBar:\r\n\r\n", headers.ToString());
    }

    TEST(HttpRequestHeaders, SetHeaderIfMissing)
    {
        HttpRequestHeaders headers;
        headers.SetHeaderIfMissing("Foo", "Bar");
        EXPECT_EQ("Foo: Bar\r\n\r\n", headers.ToString());
        headers.SetHeaderIfMissing("Foo", "Baz");
        EXPECT_EQ("Foo: Bar\r\n\r\n", headers.ToString());
    }

    TEST(HttpRequestHeaders, RemoveHeader)
    {
        HttpRequestHeaders headers;
        headers.SetHeader("Foo", "bar");
        headers.RemoveHeader("Foo");
        EXPECT_EQ("\r\n", headers.ToString());
    }

    TEST(HttpRequestHeaders, RemoveHeaderMissingHeader)
    {
        HttpRequestHeaders headers;
        headers.SetHeader("Foo", "bar");
        headers.RemoveHeader("Bar");
        EXPECT_EQ("Foo: bar\r\n\r\n", headers.ToString());
    }

    TEST(HttpRequestHeaders, RemoveHeaderCaseInsensitive)
    {
        HttpRequestHeaders headers;
        headers.SetHeader("Foo", "bar");
        headers.SetHeader("All-Your-Base", "Belongs To Chrome");
        headers.RemoveHeader("foo");
        EXPECT_EQ("All-Your-Base: Belongs To Chrome\r\n\r\n", headers.ToString());
    }

    TEST(HttpRequestHeaders, AddHeaderFromString)
    {
        HttpRequestHeaders headers;
        headers.AddHeaderFromString("Foo: bar");
        EXPECT_EQ("Foo: bar\r\n\r\n", headers.ToString());
    }

    TEST(HttpRequestHeaders, AddHeaderFromStringNoLeadingWhitespace)
    {
        HttpRequestHeaders headers;
        headers.AddHeaderFromString("Foo:bar");
        EXPECT_EQ("Foo: bar\r\n\r\n", headers.ToString());
    }

    TEST(HttpRequestHeaders, AddHeaderFromStringMoreLeadingWhitespace)
    {
        HttpRequestHeaders headers;
        headers.AddHeaderFromString("Foo: \t  \t  bar");
        EXPECT_EQ("Foo: bar\r\n\r\n", headers.ToString());
    }

    TEST(HttpRequestHeaders, AddHeaderFromStringTrailingWhitespace)
    {
        HttpRequestHeaders headers;
        headers.AddHeaderFromString("Foo: bar  \t  \t   ");
        EXPECT_EQ("Foo: bar\r\n\r\n", headers.ToString());
    }

    TEST(HttpRequestHeaders, AddHeaderFromStringLeadingTrailingWhitespace)
    {
        HttpRequestHeaders headers;
        headers.AddHeaderFromString("Foo: \t    bar\t       ");
        EXPECT_EQ("Foo: bar\r\n\r\n", headers.ToString());
    }

    TEST(HttpRequestHeaders, AddHeaderFromStringWithEmptyValue)
    {
        HttpRequestHeaders headers;
        headers.AddHeaderFromString("Foo:");
        EXPECT_EQ("Foo:\r\n\r\n", headers.ToString());
    }

    TEST(HttpRequestHeaders, AddHeaderFromStringWithWhitespaceValue)
    {
        HttpRequestHeaders headers;
        headers.AddHeaderFromString("Foo: ");
        EXPECT_EQ("Foo:\r\n\r\n", headers.ToString());
    }

    TEST(HttpRequestHeaders, MergeFrom)
    {
        HttpRequestHeaders headers;
        headers.SetHeader("A", "A");
        headers.SetHeader("B", "B");

        HttpRequestHeaders headers2;
        headers2.SetHeader("B", "b");
        headers2.SetHeader("C", "c");
        headers.MergeFrom(headers2);
        EXPECT_EQ("A: A\r\nB: b\r\nC: c\r\n\r\n", headers.ToString());
    }

    TEST(HttpRequestHeaders, CopyFrom)
    {
        HttpRequestHeaders headers;
        headers.SetHeader("A", "A");
        headers.SetHeader("B", "B");

        HttpRequestHeaders headers2;
        headers2.SetHeader("B", "b");
        headers2.SetHeader("C", "c");
        headers.CopyFrom(headers2);
        EXPECT_EQ("B: b\r\nC: c\r\n\r\n", headers.ToString());
    }

    TEST(HttpRequestHeaders, ToNetLogParamAndBackAgain)
    {
        HttpRequestHeaders headers;
        headers.SetHeader("B", "b");
        headers.SetHeader("A", "a");
        std::string request_line("GET /stuff");

        std::unique_ptr<base::Value> event_param(headers.NetLogCallback(
            &request_line, NetLogCaptureMode::IncludeCookiesAndCredentials()));
        HttpRequestHeaders headers2;
        std::string request_line2;

        ASSERT_TRUE(HttpRequestHeaders::FromNetLogParam(event_param.get(),
            &headers2,
            &request_line2));
        EXPECT_EQ(request_line, request_line2);
        EXPECT_EQ("B: b\r\nA: a\r\n\r\n", headers2.ToString());
    }

} // namespace

} // namespace net
